package com.tomdd.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.tomdd.mapper.UserMapper;
import com.tomdd.model.entity.User;
import com.tomdd.model.resp.Result;
import com.tomdd.service.UserService;
import com.tomdd.utils.JwtUtil;
import com.tomdd.utils.Md5Util;
import com.tomdd.utils.ThreadLocalUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
* @author zhongxu
* @description 针对表【user(用户表)】的数据库操作Service实现
* @createDate 2024-07-17 21:27:40
*/
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User>
    implements UserService{

    @Autowired
    private StringRedisTemplate stringRedisTemplate;



    @Override
    public User findByUserName(String username) {
        LambdaQueryWrapper<User> queryWrapper = Wrappers.lambdaQuery(User.class)
                .eq(User::getUsername, username);
        return baseMapper.selectOne(queryWrapper);
    }

    @Override
    public Result register(String username, String password) {
        // 查询用户
        User u = findByUserName(username);
        if (u == null) {
            // 没有占用
            // 注册
            //加密
            String md5String = Md5Util.getMD5String(password);
            User user = new User();
            user.setUsername(username);
            user.setPassword(md5String);
            baseMapper.insert(user);
            return Result.success();
        } else {
            // 占用
            return Result.error("用户名已被占用");
        }
    }

    @Override
    public Result<String> login(String username, String password) {
        //根据用户名查询用户
        User loginUser = findByUserName(username);
        //判断该用户是否存在
        if (loginUser == null) {
            return Result.error("用户名错误");
        }
        //判断密码是否正确  loginUser对象中的password是密文
        if (Md5Util.getMD5String(password).equals(loginUser.getPassword())) {
            //登录成功
            Map<String, Object> claims = new HashMap<>();
            claims.put("id", loginUser.getId());
            claims.put("username", loginUser.getUsername());
            String token = JwtUtil.genToken(claims);
            //把token存储到redis中
            ValueOperations<String, String> operations = stringRedisTemplate.opsForValue();
            //建议: key=userId:username
            //
            // value=token
            operations.set(token, token, 1, TimeUnit.HOURS);
            return Result.success(token);
        }
        return Result.error("密码错误");
    }

    @Override
    public Result updateAvatar(String avatarUrl) {
        ////token 放入用户id,和用户名，拦截器解析token
        //ThreadLocal中放入用户Map<String,Object> key:id,username
        Map<String,Object> map = ThreadLocalUtil.get();
        Integer id = (Integer) map.get("id");
        User user = new User();
        user.setId(id);
        user.setUserPic(avatarUrl);
        updateById(user);
        return Result.success();
    }

    @Override
    public Result updatePwd(Map<String, String> params, String token) {
        //1.校验参数
        String oldPwd = params.get("old_pwd");
        String newPwd = params.get("new_pwd");
        String rePwd = params.get("re_pwd");

        if (!StringUtils.hasLength(oldPwd) || !StringUtils.hasLength(newPwd) || !StringUtils.hasLength(rePwd)) {
            return Result.error("缺少必要的参数");
        }

        //原密码是否正确
        //调用userService根据用户名拿到原密码,再和old_pwd比对
        Map<String,Object> map = ThreadLocalUtil.get();
        String username = (String) map.get("username");
        User loginUser = findByUserName(username);
        if (!loginUser.getPassword().equals(Md5Util.getMD5String(oldPwd))){
            return Result.error("原密码填写不正确");
        }

        //newPwd和rePwd是否一样
        if (!rePwd.equals(newPwd)){
            return Result.error("两次填写的新密码不一样");
        }

        //2.调用service完成密码更新
        Integer id = (Integer) map.get("id");
        newPwd = Md5Util.getMD5String(newPwd);
        baseMapper.updatePwd(newPwd,id);
        //删除redis中对应的token
        ValueOperations<String, String> operations = stringRedisTemplate.opsForValue();
        operations.getOperations().delete(token);
        return Result.success();
    }
}




